package com.im.service.config;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.context.model.SaRequest;
import cn.dev33.satoken.interceptor.SaInterceptor;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.util.StrUtil;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

/**
 * @author liuzhiqiang
 */
@Configuration
public class SaTokenConfigure implements WebMvcConfigurer {
    /**
     * 注册 Sa-Token 拦截器，打开注解式鉴权功能
     *
     * @param registry
     */
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        // 注册 Sa-Token 拦截器，打开注解式鉴权功能
        // 注册 Sa-Token 拦截器，校验规则为 StpUtil.checkLogin() 登录校验。
        registry
                .addInterceptor(new SaInterceptor(handle -> {
                    SaRequest saRequest = SaHolder.getRequest();
                    if (!StrUtil.equals(saRequest.getMethod(), "OPTIONS")) {
                        StpUtil.checkLogin();
                    }
                }))
                .addPathPatterns("/**")
                .excludePathPatterns(
                        "/imChat/syncUserCache",
                        "/init/**",
                        "/login",
                        "/error",
                        "/druid/**",
                        "/doc.html",
                        "/webjars/**",
                        "/favicon.ico",
                        "/swagger-resources",
                        "/addressBook/syncOrgUser",
                        "/oss/rongHui/**",
                        "/avatar/**"
                );
    }
}
